#include <linux/types.h>
#include <linux/kernel.h>
#include <linux/delay.h>
#include <linux/ide.h>
#include <linux/init.h>
#include <linux/module.h>
#include <linux/errno.h>
#include <linux/gpio.h>
#include <linux/cdev.h>
#include <linux/device.h>
//#include <asm/mach/map.h>
#include <asm/uaccess.h>
#include <asm/io.h>
/***************************************************************
文件名   	: newchrled.c
版本      	: V1.0
描述      	: LED驱动文件。
其他      	: 无
日志      	: 初版V1.0 2023/12/02 
***************************************************************/
#define NEWCHRLED_CNT       	1           		/* 设备号个数 	*/
#define NEWCHRLED_NAME      	"newchrled" 		/* 名字 		*/
#define LEDOFF               	0           		/* 关灯 		*/
#define LEDON                	1           		/* 开灯 		*/

#define SYS_GRF_BASE						(0xFDC60000)
#define GRF_GPIO3D_IOMUX_L					(SYS_GRF_BASE + 0x0058)
#define GRF_GPIO3D_DS_0						(SYS_GRF_BASE + 0X02B0)

#define GPIO3_BASE						(0xFE760000)
#define GPIO3_SWPORT_DR_H				(GPIO3_BASE + 0X0004)
#define GPIO3_SWPORT_DDR_H				(GPIO3_BASE + 0X000C)

/* 映射后的寄存器虚拟地址指针 */
static void __iomem *GRF_GPIO3D_IOMUX_L_PI;
static void __iomem *GRF_GPIO3D_DS_0_PI;
static void __iomem *GPIO3_SWPORT_DR_H_PI;
static void __iomem *GPIO3_SWPORT_DDR_H_PI;

/* newchrled设备结构体 */
struct newchrled_dev{
    dev_t devid;            	/* 设备号     	*/
    struct cdev cdev;       	/* cdev     	*/
    struct class *class;    	/* 类      		*/
    struct device *device;  	/* 设备    		*/
    int major;              	/* 主设备号   	*/
    int minor;              	/* 次设备号   	*/
};

struct newchrled_dev newchrled; /* led设备 */

/*
 * @description  	: LED打开/关闭
 * @param - sta 	: LEDON(0) 打开LED，LEDOFF(1) 关闭LED
 * @return       	: 无
 */
void led_switch(u8 sta)
{
	u32 val = 0;
	if(sta == LEDON) {
		val = readl(GPIO3_SWPORT_DR_H_PI);
		val &= ~(0X1 << 8); /* bit0 清零*/
		val |= ((0X1 << 24) | (0X1 << 8));	/* bit24 置1，允许写bit8，
	 					   				   bit8，高电平*/
		writel(val, GPIO3_SWPORT_DR_H_PI);
	}else if(sta == LEDOFF) { 
		val = readl(GPIO3_SWPORT_DR_H_PI);
		val &= ~(0X1 << 8); /* bit0 清零*/
		val |= ((0X1 << 24) & (~(0X1 << 8)));	/* bit24 置1，允许写bit8，
	 					   				   bit8，低电平	*/
		writel(val, GPIO3_SWPORT_DR_H_PI);
	}	
}


/*
 * @description 	: 物理地址映射
 * @return        	: 无
 */
void led_remap(void)
{
  	GRF_GPIO3D_IOMUX_L_PI = ioremap(GRF_GPIO3D_IOMUX_L, 4);
	GRF_GPIO3D_DS_0_PI = ioremap(GRF_GPIO3D_DS_0, 4);
	GPIO3_SWPORT_DR_H_PI = ioremap(GPIO3_SWPORT_DR_H, 4);
	GPIO3_SWPORT_DDR_H_PI = ioremap(GPIO3_SWPORT_DDR_H, 4);
}


/*
 * @description 	: 取消映射
 * @return       	: 无
 */
void led_unmap(void)
{
	/* 取消映射 */
	iounmap(GRF_GPIO3D_IOMUX_L_PI);
	iounmap(GRF_GPIO3D_DS_0_PI);
	iounmap(GPIO3_SWPORT_DR_H_PI);
	iounmap(GPIO3_SWPORT_DDR_H_PI);
}


/*
 * @description  	: 打开设备
 * @param – inode	: 传递给驱动的inode
 * @param - filp 	: 设备文件，file结构体有个叫做private_data的成员变量
 *                    一般在open的时候将private_data指向设备结构体。
 * @return       	: 0 成功;其他 失败
 */
static int led_open(struct inode *inode, struct file *filp)
{
    filp->private_data = &newchrled; /* 设置私有数据 */
    return 0;
}

/*
 * @description  	: 从设备读取数据 
 * @param - filp 	: 要打开的设备文件(文件描述符)
 * @param - buf  	: 返回给用户空间的数据缓冲区
 * @param - cnt  	: 要读取的数据长度
 * @param - offt 	: 相对于文件首地址的偏移
 * @return        	: 读取的字节数，如果为负值，表示读取失败
 */
static ssize_t led_read(struct file *filp, char __user *buf, 
size_t cnt, loff_t *offt)
{
    return 0;
}

/*
 * @description  	: 向设备写数据 
 * @param – filp	: 设备文件，表示打开的文件描述符
 * @param - buf 	: 要写给设备写入的数据
 * @param - cnt  	: 要写入的数据长度
 * @param - offt 	: 相对于文件首地址的偏移
 * @return        	: 写入的字节数，如果为负值，表示写入失败
 */
static ssize_t led_write(struct file *filp, const char __user *buf, 
size_t cnt, loff_t *offt)
{
    int retvalue;
    unsigned char databuf[1];
    unsigned char ledstat;

    retvalue = copy_from_user(databuf, buf, cnt);
    if(retvalue < 0) {
        printk("kernel write failed!\r\n");
        return -EFAULT;
    }

    ledstat = databuf[0];       /* 获取状态值 */

    if(ledstat == LEDON) {  
        led_switch(LEDON);      /* 打开LED灯 */
    } else if(ledstat == LEDOFF) {
        led_switch(LEDOFF);     /* 关闭LED灯 */
    }
    return 0;
}

/*
 * @description   : 关闭/释放设备
 * @param - filp 	: 要关闭的设备文件(文件描述符)
 * @return        	: 0 成功;其他 失败
 */
static int led_release(struct inode *inode, struct file *filp)
{
    return 0;
}

/* 设备操作函数 */
static struct file_operations newchrled_fops = {
    .owner = THIS_MODULE,
    .open = led_open,
    .read = led_read,
    .write = led_write,
    .release =  led_release,
};

/*
 * @description 	: 驱动出口函数
 * @param       	: 无
 * @return      	: 无
 */
static int __init led_init(void)
{
    u32 val = 0;
    int ret;

    /* 初始化LED */
    /* 1、寄存器地址映射 */
    led_remap();

	/* 2、设置GPIO3_D0为GPIO功能。*/
	val = readl(GRF_GPIO3D_IOMUX_L_PI);
	val &= ~(0X7 << 0);	/* bit2:0，清零 */
	val |= ((0X7 << 16) & (~(0X1 << 0)));	/* bit18:16 置1，允许写bit2:0，
	 					   				   bit2:0：0，用作GPIO3_D0	*/
	writel(val, GRF_GPIO3D_IOMUX_L_PI);

	/* 3、设置GPIO3_D0驱动能力为level5 */
	val = readl(GRF_GPIO3D_DS_0_PI);
	val &= ~(0X3F << 0);	/* bit5:0清零*/
	val |= ((0X3F << 16) | (0X3F << 0));	/* bit21:16 置1，允许写bit5:0，
	 					   				   bit5:0：0，用作GPIO3_D0	*/
	writel(val, GRF_GPIO3D_DS_0_PI);

	/* 4、设置GPIO3_D0为输出 */
	val = readl(GPIO3_SWPORT_DDR_H_PI);
	val &= ~(0X1 << 8); /* bit8 清零*/
	val |= ((0X1 << 24) | (0X1 << 8));	/* bit24 置1，允许写bit8，
	 					   				   bit8，高电平	*/
	writel(val, GPIO3_SWPORT_DDR_H_PI);

    /* 注册字符设备驱动 */
    /* 1、创建设备号 */
    if (newchrled.major) {      /*  定义了设备号 */
        newchrled.devid = MKDEV(newchrled.major, 0);
        ret = register_chrdev_region(newchrled.devid, NEWCHRLED_CNT, NEWCHRLED_NAME);
         if(ret < 0)
         {
             pr_err("cannot register %s char driver [ret=%d]\n",NEWCHRLED_NAME, NEWCHRLED_CNT);
             goto fail_map;
         }
     } else 
     {                        /* 没有定义设备号 */
         ret = alloc_chrdev_region(&newchrled.devid, 0, NEWCHRLED_CNT, NEWCHRLED_NAME);  /* 申请设备号 */
         if(ret < 0) 
         {
             pr_err("%s Couldn't alloc_chrdev_region, ret=%d\r\n", NEWCHRLED_NAME, ret);
             goto fail_map;
         }
         newchrled.major = MAJOR(newchrled.devid);/* 获取主设备号 */
         newchrled.minor = MINOR(newchrled.devid);/* 获取次设备号 */
     }
     printk("newcheled major=%d,minor=%d\r\n",newchrled.major, newchrled.minor); 
     
     /* 2、初始化cdev */
     newchrled.cdev.owner = THIS_MODULE;
     cdev_init(&newchrled.cdev, &newchrled_fops);
     
     /* 3、添加一个cdev */
     ret = cdev_add(&newchrled.cdev, newchrled.devid, NEWCHRLED_CNT);
     if(ret < 0)
         goto del_unregister;
         
     /* 4、创建类 */
     newchrled.class = class_create(THIS_MODULE, NEWCHRLED_NAME);
     if (IS_ERR(newchrled.class)) {
         goto del_cdev;
     }
 
     /* 5、创建设备 */
     newchrled.device = device_create(newchrled.class, NULL, newchrled.devid, NULL, NEWCHRLED_NAME);
     if (IS_ERR(newchrled.device)) {
         goto destroy_class;
     }
     
     return 0;
 
destroy_class:
    class_destroy(newchrled.class);
del_cdev:
    cdev_del(&newchrled.cdev);
del_unregister:
    unregister_chrdev_region(newchrled.devid, NEWCHRLED_CNT);
fail_map:
    led_unmap();
    return -EIO;
}

/*
 * @description 	: 驱动出口函数
 * @param       	: 无
 * @return      	: 无
 */
static void __exit led_exit(void)
{
    /* 取消映射 */
    led_unmap();
   
    /* 注销字符设备驱动 */
    cdev_del(&newchrled.cdev);/*  删除cdev */
    unregister_chrdev_region(newchrled.devid, NEWCHRLED_CNT);

    device_destroy(newchrled.class, newchrled.devid);
    class_destroy(newchrled.class);
}

module_init(led_init);
module_exit(led_exit);
MODULE_LICENSE("GPL");
MODULE_AUTHOR("ZOROE");
MODULE_INFO(intree, "Y");